leap year program in python using function